package com.sky.aspect;

import com.sky.annotation.AutoFill;
import com.sky.constant.AutoFillConstant;
import com.sky.context.BaseContext;
import com.sky.enumeration.OperationType;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.Date;

/**
 * 切面类：实现公共字段自动填充处理逻辑
 */
@Slf4j
@Component
@Aspect // 切面类
public class AutoFillAspect {

    /**
     * 切入点 :拦截mapper下的所有接口 以及 注解 AutoFill
     */
    @Pointcut("execution (* com.sky.mapper.*.*(..)) && @annotation(com.sky.annotation.AutoFill)")
    public void autoFillPoinCut() {
    }


    // 前置通知  通知方法在目标方法前执行
    @Before("com.sky.aspect.AutoFillAspect.autoFillPoinCut()")
    public void autoFill(JoinPoint joinPoint){

        log.info("before ......");


        //1、获取操作方法的类型：新增 修改
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();// 方法签名对象
        AutoFill annotation = signature.getMethod().getAnnotation(AutoFill.class);// 获取方法上的注解对象
        OperationType value = annotation.value();

        //2、获取被拦截的方法的参数 -实体对象
        Object[] args = joinPoint.getArgs();
        if (args == null || args.length < 1) {
            return ;
        }
        Object entity = args[0];

            if (value.equals(OperationType.INSERT)) {
                try {
                    // 反射获得用户id的赋值方法
                    Method setCreateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_USER, Long.class);
                    Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, Long.class);
                    Method setCreateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_CREATE_TIME, Date.class);
                    Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, Date.class);
                    setCreateTime.invoke(entity, new Date());
                    setUpdateTime.invoke(entity, new Date());
                    setCreateUser.invoke(entity, BaseContext.getCurrentId());
                    setUpdateUser.invoke(entity, BaseContext.getCurrentId());
                } catch (Exception e) {
                    e.printStackTrace();
                }

            } else if (value.equals(OperationType.UPDATE)) {
                try {
                    // 反射获得用户id的赋值方法
                    Method setUpdateUser = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_USER, Long.class);
                    Method setUpdateTime = entity.getClass().getDeclaredMethod(AutoFillConstant.SET_UPDATE_TIME, Date.class);
                    setUpdateTime.invoke(entity, new Date());
                    setUpdateUser.invoke(entity, BaseContext.getCurrentId());
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
    }
}
